<!DOCTYPE html>
<meta charset="utf-8">
<title>SANKEY Experiment</title>
<style>
 
.node rect {
  cursor: move;
  fill-opacity: .9;
  shape-rendering: crispEdges;
}
 
.node text {
  pointer-events: none;
  text-shadow: 0 1px 0 #fff;
}
 
.link {
  fill: none;
  stroke: #000;
  stroke-opacity: .2;
}
 
.link:hover {
  stroke-opacity: .5;
}
 
</style>
<body>
 
<p id="chart">
 
<script src="http://d3js.org/d3.v3.js"></script>
<script src="sankey.js"></script>
<script>

var units = "Units";

var margin = {top: 10, right: 10, bottom: 10, left: 10},
    width = 1024 - margin.left - margin.right,
    height = 768 - margin.top - margin.bottom;
 
var formatNumber = d3.format(",.0f"),    // zero decimal places
    format = function(d) { return formatNumber(d) + " " + units; },
    color = d3.scale.category20();

// append the svg canvas to the page
var svg = d3.select("#chart").append("svg")
    .attr("width", width + margin.left + margin.right)
    .attr("height", height + margin.top + margin.bottom)
  .append("g")
    .attr("transform", 
          "translate(" + margin.left + "," + margin.top + ")");

// Set the sankey diagram properties
var sankey = d3.sankey()
    .nodeWidth(36)
    .nodePadding(10)
    .size([width, height]);
 
var path = sankey.link();

// load the data with d3.csv instead of d3.json
//for another much simpler example uncomment the below
// d3.csv("d3noob_example.csv", function(error, data) {
d3.tsv("data/out_qc2_agg10.tsv", function(error, data) {
 		//set up graph in same style as original example but empty
  		graph = {"nodes" : [], "links" : []};
				//out_qc1_agg10.tsv
                data.forEach(function (d) {
                	if((+d.count_actions) >= 10000) {
                    graph.nodes.push({ "name": d.from_lang+">" });
                    graph.nodes.push({ "name": ">"+d.to_lang });
                    graph.links.push({ "source": d.from_lang+">", "target": ">"+d.to_lang, "value": +d.count_actions });
                    }
                });
				/*
                //twitter_gln_links_with_over_expression.tsv
                data.forEach(function (d) {
                	if((+d["Common Users"]) >= 10000) {
                    graph.nodes.push({ "name": d.Source+">" });
                    graph.nodes.push({ "name": ">"+d.Target });
                    graph.links.push({ "source": d.Source+">", "target": ">"+d.Target, "value": +d["Common Users"] });
                    }
                });
                //books_edges.tsv
                data.forEach(function (d) {
                	if((+d["Coocurrences"]) >= 2000) {
                    graph.nodes.push({ "name": d.SourceLanguageCode+">" });
                    graph.nodes.push({ "name": ">"+d.TargetLanguageCode });
                    graph.links.push({ "source": d.SourceLanguageCode+">", "target": ">"+d.TargetLanguageCode, "value": +d["Coocurrences"] });
                    }
                });
                */
                //thanks Mike Bostock https://groups.google.com/d/msg/d3-js/pl297cFtIQk/Eso4q_eBu1IJ
                //this handy little function returns only the distinct / unique nodes
                graph.nodes = d3.keys(d3.nest()
                         .key(function (d) { return d.name; })
                         .map(graph.nodes));

                //it appears d3 with force layout wants a numeric source and target
                //so loop through each link replacing the text with its index from node
                graph.links.forEach(function (d, i) {
                    graph.links[i].source = graph.nodes.indexOf(graph.links[i].source);
                    graph.links[i].target = graph.nodes.indexOf(graph.links[i].target);
                });

                //now loop through each nodes to make nodes an array of objects rather than an array of strings
                graph.nodes.forEach(function (d, i) {
                    graph.nodes[i] = { "name": d };
                });
 
  sankey
      .nodes(graph.nodes)
      .links(graph.links)
      .layout(32);
 
// add in the links
  var link = svg.append("g").selectAll(".link")
      .data(graph.links)
    .enter().append("path")
      .attr("class", "link")
      .attr("d", path)
      .style("stroke-width", function(d) { return Math.max(1, d.dy); })
      .sort(function(a, b) { return b.dy - a.dy; });
 
// add the link titles
  link.append("title")
        .text(function(d) {
      	return d.source.name + " → " + 
                d.target.name + "\n" + format(d.value); });
 
// add in the nodes
  var node = svg.append("g").selectAll(".node")
      .data(graph.nodes)
    .enter().append("g")
      .attr("class", "node")
      .attr("transform", function(d) { 
		  return "translate(" + d.x + "," + d.y + ")"; })
    .call(d3.behavior.drag()
      .origin(function(d) { return d; })
      .on("dragstart", function() { 
		  this.parentNode.appendChild(this); })
      .on("drag", dragmove));
 
// add the rectangles for the nodes
  node.append("rect")
      .attr("height", function(d) { return d.dy; })
      .attr("width", sankey.nodeWidth())
      .style("fill", function(d) { 
		  return d.color = color(d.name.replace(/ .*/, "")); })
      .style("stroke", function(d) { 
		  return d3.rgb(d.color).darker(2); })
    .append("title")
      .text(function(d) { 
		  return d.name + "\n" + format(d.value); });
 
// add in the title for the nodes
  node.append("text")
      .attr("x", -6)
      .attr("y", function(d) { return d.dy / 2; })
      .attr("dy", ".35em")
      .attr("text-anchor", "end")
      .attr("transform", null)
      .text(function(d) { return d.name; })
    .filter(function(d) { return d.x < width / 2; })
      .attr("x", 6 + sankey.nodeWidth())
      .attr("text-anchor", "start");
 
// the function for moving the nodes
  function dragmove(d) {
    d3.select(this).attr("transform", 
        "translate(" + (
        	   d.x = Math.max(0, Math.min(width - d.dx, d3.event.x))
        	) + "," + (
                   d.y = Math.max(0, Math.min(height - d.dy, d3.event.y))
            ) + ")");
    sankey.relayout();
    link.attr("d", path);
  }
});
 
</script>
 
</body>
</html>
